Skip to content

[WIP] Support ordered first/last groupby aggregations in cudf-polars - #23600

Draft
rjzamora wants to merge 19 commits into
NVIDIA:mainfrom
rjzamora:groupby-sortby-agg
Draft

[WIP] Support ordered first/last groupby aggregations in cudf-polars#23600
rjzamora wants to merge 19 commits into
NVIDIA:mainfrom
rjzamora:groupby-sortby-agg

Conversation

@rjzamora

@rjzamora rjzamora commented Aug 8, 2026

Copy link
Copy Markdown
Contributor

Closes #23592

Adds cudf-polars support for grouped sort_by(...).first() and sort_by(...).last() aggregations, including streaming decomposition for multi-partition execution.

This enables expressions like:

df.group_by("g").agg(
    pl.col("x").sort_by("t").first().alias("open"),
    pl.col("x").sort_by("t").last().alias("close"),
)

@rjzamora rjzamora self-assigned this Aug 8, 2026
@rjzamora rjzamora added feature request New feature or request 2 - In Progress Currently a work in progress non-breaking Non-breaking change labels Aug 8, 2026
@copy-pr-bot

copy-pr-bot Bot commented Aug 8, 2026

Copy link
Copy Markdown

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@github-actions github-actions Bot added Python Affects Python cuDF API. cudf-polars Issues specific to cudf-polars labels Aug 8, 2026
@rjzamora

rjzamora commented Aug 8, 2026

Copy link
Copy Markdown
Contributor Author

/ok to test

@GPUtester GPUtester moved this to In Progress in cuDF Python Aug 8, 2026
@rjzamora

Copy link
Copy Markdown
Contributor Author

/ok to test

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

2 - In Progress Currently a work in progress cudf-polars Issues specific to cudf-polars feature request New feature or request non-breaking Non-breaking change Python Affects Python cuDF API.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[FEA] Support ordered first/last grouped reductions in cuDF-Polars

2 participants